how to write merge sort in python